
#ifndef __EDNN_DENSE_H__
#define __EDNN_DENSE_H__



#include "ednn_layers.h"



struct ednn_layer_dense
{
    struct ednn_layer   super;

    struct ednn_tensor *weights;
    struct ednn_tensor *bias;

    ednn_qfmt_parm_t   *bias_shift;
    ednn_qfmt_parm_t   *output_shift;
};

typedef struct ednn_layer_dense ednn_layer_dense_t;



struct ednn_layer *ednn_dense_layer(ednn_size_t out_features);
#define Dense(out_features) ednn_dense_layer(out_features)

#endif /* __EDNN_DENSE_H__ */
